package com.sea.data.controller;

import com.codecrane.core.util.DateUtils;
import com.codecrane.core.util.ExcelUtil;
import com.codecrane.core.web.Ajax;
import com.codecrane.core.web.AjaxReturn;
import com.codecrane.core.web.AjaxTableReturn;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.sea.data.entity.Order;
import com.sea.data.entity.SalesForecast;
import com.sea.data.service.OrderService;
import com.sea.data.service.SalesForecastService;
import com.sea.physical.entity.CreateDistributeSku;
import com.sea.product.service.CategoryService;
import com.sea.product.service.ChannelService;
import com.sea.supply.entity.Produce;
import com.sea.supply.service.ProduceService;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.dao.DuplicateKeyException;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

@Controller
@RequestMapping(value = "/sea/data/sales")
public class SalesForecastController {

    @Autowired
    private SalesForecastService salesForecastService;
    @Autowired
    private ChannelService channelService;

    @RequestMapping("/reset")
    @ResponseBody
    public AjaxReturn reset(Model model) {
        AjaxReturn result = Ajax.fail().setMsg("保存失败！");
        try {
            salesForecastService.resetSalesForecast();
            result.setOk(true).setMsg("清空成功");
        }catch (Exception e) {
            e.printStackTrace();
            result.setMsg("系统异常,请联系管理员");
            return result;
        }
        return result;
    }



    /**
     * 管理主页面
     * @param model
     * @return
     */
    @RequestMapping("/main")
    public String main(Model model) {
        //渠道
        model.addAttribute("channel", channelService.findByCnd(null));
        return "/sea/data/sales/main";
    }

    @ResponseBody
    @RequestMapping("/findByCnd")
    public AjaxTableReturn findByCnd(@RequestParam(value = "limit", defaultValue = "10") int pagesize,
                                     @RequestParam(value = "offset", defaultValue = "0") int start,
                                     SalesForecast salesForecast) {
        start = start > 0 ? start / pagesize + 1 : 1;
        PageHelper.startPage(start, pagesize);
        List<SalesForecast> list = salesForecastService.findByCnd(salesForecast);
        PageInfo<SalesForecast> page = new PageInfo<>(list);
        return Ajax.tableDataOk(page.getTotal(), page.getList());
    }

    @RequestMapping("/import")
    @ResponseBody
    public AjaxReturn importClubDo(Model model, @RequestParam("productMember")String productMember, MultipartFile file) {
        AjaxReturn result = Ajax.fail().setMsg("保存失败！");
        try {
            if (null == file || file.isEmpty()) {
                return result.setMsg("请上传文件！！！");
            } else {
                String fileName = file.getOriginalFilename();
                if (!(fileName.contains(".xls"))) {
                    return result.setMsg("只能上传excel文件！！！");
                }
                salesForecastService.importSku(file.getInputStream(), fileName,productMember);
                result.setOk(true);
                result.setMsg("上传成功！！");
            }
        } catch (Exception e) {
            e.printStackTrace();
            result.setMsg(e.getMessage());
            return result;
        }
        return result;
    }

    @RequestMapping("/importStock")
    @ResponseBody
    public AjaxReturn importStock(Model model, @RequestParam("salesChannelName")String salesChannelName, @RequestParam("salesDate")String salesDate, MultipartFile importSession) {
        AjaxReturn result = Ajax.fail().setMsg("保存失败！");
        try {
            if (null == importSession || importSession.isEmpty()) {
                return result.setMsg("请上传文件！！！");
            } else {
                String fileName = importSession.getOriginalFilename();
                if (!(fileName.contains(".xls"))) {
                    return result.setMsg("只能上传excel文件！！！");
                }
                salesForecastService.importSkuStock(importSession.getInputStream(), fileName, salesChannelName,salesDate);
                result.setOk(true);
                result.setMsg("上传成功！！");
            }
        }catch (RuntimeException e){
            e.printStackTrace();
            result.setMsg(e.getMessage());
            return result;
        } catch (Exception e) {
            e.printStackTrace();
            result.setMsg("系统异常,请联系管理员");
            return result;
        }
        return result;
    }


    @ResponseBody
    @RequestMapping(value = "/export")
    public void export(HttpServletResponse response, SalesForecast salesForecast)
    {

        List<SalesForecast> list = salesForecastService.findByCnd(salesForecast);
        String[] title = {"Listing","渠道","SKU","ASIN","日期","日均销量","海运入库","采购入库","期末库存","海运入库(海外仓)","采购入库(海外仓)","可售库存"};
        String fileName = "销量预测表" + DateUtils.dateToString(new Date(),"yyyyMMdd") + ".xls";
        String sheetName = "销量预测表";
        String[][] content = new String[list.size()][12];
        for (int i = 0; i < list.size(); i++) {
            content[i] = new String[title.length];
            SalesForecast obj = list.get(i);
            content[i][0] = obj.getSalesListing();
            content[i][1] = obj.getSalesChannelName();
            content[i][2] = obj.getSalesSku();
            content[i][3] = obj.getSalesAsin();
            content[i][4] = obj.getSalesDate();
            content[i][5] = obj.getSalesNumber()+"";
            content[i][6] = obj.getSalesPhysicalNumber()+"";
            content[i][7] = obj.getSalesProduceNumber()+"";
            content[i][8] = obj.getSalesStock()+"";
            content[i][9] = obj.getSalesStockPhysicalNumber()+"";
            content[i][10] = obj.getSalesStockProduceNumber()+"";
            content[i][11] = obj.getSalesSellNumber()+"";
        }
        //创建HSSFWorkbook
        HSSFWorkbook wb = ExcelUtil.getHSSFWorkbook(sheetName, title, content, null);
        //响应到客户端
        try {
            this.setResponseHeader(response, fileName);
            OutputStream os = response.getOutputStream();
            wb.write(os);
            os.flush();
            os.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }


    //发送响应流方法
    public void setResponseHeader(HttpServletResponse response, String fileName) {
        try {
            try {
                fileName = new String(fileName.getBytes(),"ISO8859-1");
            } catch (UnsupportedEncodingException e) {
                e.printStackTrace();
            }
            response.setContentType("application/octet-stream;charset=ISO8859-1");
            response.setHeader("Content-Disposition", "attachment;filename="+ fileName);
            response.addHeader("Pargam", "no-cache");
            response.addHeader("Cache-Control", "no-cache");
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    }

}
